# -*- coding: utf-8 -*-
from pymongo import Connection

MONGO_SERVER = 'beta.mana.vn'
MONGO_PORT = 27017

cat = ['Văn hóa - Xã hội', 'Thế giới', 'Thể thao', 
       'Giải trí', 'Giáo dục', 'Pháp luật', 'Sức khỏe - Giới tính', 
       'Khoa học', 'Công nghệ số', 'Làm đẹp - Thời trang', 
       'Kinh doanh', 'Tiêu dùng', 'Ẩm thực', 'Chuyện lạ', 'Tâm sự']

connection = Connection(MONGO_SERVER, MONGO_PORT)
db = connection['tintuc_v2']
collection = db['category']

r = 50
node_id = 0
left = 1
right = 1
try:
    crRgt = collection.find({}, {'rgt': 1, 'node_id': 1}).sort('rgt', -1).limit(1)
    if crRgt != None:
        node_id = crRgt[0]['node_id']
        left = crRgt[0]['rgt'] + 1
        right = left
except:
    pass

for i in range(len(cat)):
    if not collection.find_one({'data': cat[i]}):
        if left < right: left = right + 1
        if node_id == 0:
            node_id = i + 5
        else:
            node_id += 5
        root_id = node_id
        level = 1
        right = left + r
        collection.save({
            'data': cat[i],
            'level': 0,
            'lft': left,
            'rgt': right,
            'node_id': node_id,
            'root_id': root_id
        })
    
print 'make category done'

import os
os._exit(1)
